1783D - Different Arrays - CodeForces Solution

def main():

    MOD = 998244353
    n = int(input())
    a = readIntArr()

        
        offset = 100000
    dp = [0] * 200000      dp[offset + a[1]] = 1
    for i in range(1, n - 1):
                dp2 = [0] * 200000
        for v in range(-90000, 90000):
            if -100000 < a[i + 1] + v < 100000:
                dp2[offset + a[i + 1] + v] = (dp2[offset + a[i + 1] + v] + dp[offset + v]) % MOD
            if v != 0 and -100000 < a[i + 1] - v < 100000:
                dp2[offset + a[i + 1] - v]
                dp2[offset + a[i + 1] - v] = (dp2[offset + a[i + 1] - v] + dp[offset + v]) % MOD
        dp = dp2
    ans = 0
    for cnts in dp:
        ans += cnts
        ans %= MOD
    print(ans)

        
    return

import sys
input=sys.stdin.buffer.readline  
def oneLineArrayPrint(arr):
    print(' '.join([str(x) for x in arr]))
def multiLineArrayPrint(arr):
    print('\n'.join([str(x) for x in arr]))
def multiLineArrayOfArraysPrint(arr):
    print('\n'.join([' '.join([str(x) for x in y]) for y in arr]))
 
def readIntArr():
    return [int(x) for x in input().split()]
 
def makeArr(defaultValFactory,dimensionArr):     dv=defaultValFactory;da=dimensionArr
    if len(da)==1:return [dv() for _ in range(da[0])]
    else:return [makeArr(dv,da[1:]) for _ in range(da[0])]
 
def queryInteractive(a, b):
    print('? {} {}'.format(a, b))
    sys.stdout.flush()
    return int(input())
 
def answerInteractive(ans):
    print('! {}'.format(ans))
    sys.stdout.flush()
 
inf=float('inf')
 
from math import gcd,floor,ceil
import math
 
for _abc in range(1):
    main()